import json
import os
from agent.memory import LearningMemory

def main():
    print("Hello from learning-agent!")


def initialize_existing_memory():
    """
    初始化现有学习记忆文件的掌握程度值
    """
    # 获取data目录下所有的learning_memory_*.json文件
    data_dir = "data"
    if not os.path.exists(data_dir):
        print("data目录不存在")
        return
    
    for filename in os.listdir(data_dir):
        if filename.startswith("learning_memory_") and filename.endswith(".json"):
            # 提取学生ID
            try:
                student_id = filename.split("_")[-1].split(".")[0]
                print(f"处理学生ID: {student_id}")
                
                # 加载并保存记忆，这会自动初始化掌握程度
                memory = LearningMemory(student_id)
                memory.save_memory()
                
                print(f"已为学生 {student_id} 初始化掌握程度")
            except Exception as e:
                print(f"处理文件 {filename} 时出错: {e}")


# if __name__ == "__main__":
#     # 初始化现有学习记忆文件的掌握程度
#     initialize_existing_memory()
#     print("所有学习记忆文件的掌握程度初始化完成！")
#     # 调用原来的main函数
#     main()

from dotenv import load_dotenv
# 加载.env文件中的环境变量
load_dotenv()

from ui.web_ui import create_interface

# 启动界面
if __name__ == "__main__":
    interface = create_interface()
    interface.launch(debug=True)